#!/usr/bin/env bash
CUDA_VISIBLE_DEVICES='0' python ./MTDet.py \
--dataroot '/home/sean/data/UWdevkit/snippets/2018_2.mkv' \
--model test \
--which_model_netG 'resnet_9blocks' \
--which_direction 'AtoB' \
--dataset_mode 'single' \
--norm 'batch' \
--checkpoints_dir './GAN_RS/checkpoints' \
--which_epoch 65 \
--loadSize 512 \
--fineSize 512 \
--gpu_id '0' \
--backbone 'RefineDet_VGG' \
--ssd_dim 320 \
--bn 'yes' \
--refine 'yes' \
--deform 0 \
--multihead 'no' \
--tub 5 \
--loss_hold_len 5 \
--tub_thresh 0.9 \
--tub_generate_score 0.1 \
--confidence_thresh 0.4 \
--conf_list 0.3 0.3 0.3 0.5 \
--top_k 200 \
--pool_size 3 \
--model_dir './SSD/weights040/UW/ssd512RefineBN_UWm1718ex/ssd512_UW_130.pth' \
--name 'underwater_pix2pix512_Res9Gmultibranch46D_selectDdcpL1a30lu5gan1_lsgan' \
--show_video
#--results_dir './results/test' \
#--writename 'record.avi' \
# '/home/sean/data/UWdevkit/Data/snippets/2.MP4'
# 'rtsp://192.168.1.11:554/user=admin&password=&channel=1&stream=.sdp?'
